import time
from gen_basic_old import entry_basic
from gen_extend_old import entry_extend
from merge_results import merge

from data.config import BASIC_GENERATED_TESTS_DIR, EXTEND_GENERATED_TESTS_DIR,MERGED_GENERATED_TESTS_DIR

if __name__ == '__main__':
    # qwen3-next-80b-a3b-instruct
    # qwen2.5-72b-instruct
    # qwen3-32b
    # deepseek-r1-distill-qwen-32b
    # models = ['qwen3-32b', 'qwen2.5-72b-instruct', 'deepseek-r1-distill-qwen-32b']
    # models = ['deepseek-r1-distill-qwen-32b', 'qwen2.5-72b-instruct']
    models = ['deepseek/deepseek-r1-distill-llama-70b:free']
    for model in models:
        date = time.strftime("%m%d_%H%M", time.localtime())
        
        # basic_generation_iden = entry_basic(model)
        basic_generation_iden = 'results_1106_0951_deepseek-deepseek-r1-distill-llama-70b.jsonl'
        # extend_generation_iden = entry_extend(model, basic_generation_iden)
        extend_generation_iden = 'results_1106_1004_deepseek-deepseek-r1-distill-llama-70b:free.jsonl'
        
        basic_file = f'{BASIC_GENERATED_TESTS_DIR}/{basic_generation_iden}'
        extend_file = f'{EXTEND_GENERATED_TESTS_DIR}/{extend_generation_iden}'
        output_file = f'{MERGED_GENERATED_TESTS_DIR}/merged_{date}_{model}.csv'
        merge(basic_file, extend_file, output_file)